import streamlit as st
from selected_models import get_response_from14 ,get_response_from18,get_response_from32

models = ['18','14','32']
selected_model = st.selectbox('请选择一个大模型：',models)
st.write(f'你现在选择了{selected_model}大模型')
user_question = st.text_input('请输入你的问题：')
if st.button('发送'):
    if selected_model == '18':
        response = get_response_from18(user_question)
    elif selected_model =='14':
        response = get_response_from14(user_question)
    elif selected_model == '32':
        response = get_response_from32(user_question)
    else:
        response = "未知模型"
    st.write(response)